ConSA: Controllable Sparsity in Hybrid Attention via Learnable Allocation
可控稀疏的混合注意力:通过可学习分配优化FA/SWA组合
🧭 高效LLM推理 · 混合注意力稀疏分配 ★ 深度解读
Institute of Information Engineering, Chinese Academy of Sciences · School of Cyber Security, University of Chinese Academy of Sciences · Baidu Inc.
用L0正则化与增广拉格朗日约束学习全局/局部注意力的最优分配,实现可控稀疏并揭示底层局部、中层全局的一致结构
📌 工作参考价值(4 点)
- 部署长上下文LLM(如query理解、会话摘要)时,可用ConSA动态决定哪些层/头用全局注意力、哪些用滑动窗口,在几乎不降性能的同时减少KV cache和推理计算。
- 借鉴可控稀疏思想,在推荐系统的Transformer中为不同用户行为序列(短期 vs. 长期)分配不同的注意力范围,用类似L0-拉格朗日方法学习分配。
- 增广拉格朗日约束可精确满足目标稀疏率,适合在业务中需要严格控制资源预算的场景(例如单机GPU限制)。
- 手工设计混合注意力时可参考已发现的规律:底层放局部注意力、中层放全局注意力,避免均匀交错模式。